use API provided by the user.

.gitignore

@@ -0,0 +1,3 @@
+.env
+index
+__pycache__

agent_workflow.py

@@ -9,17 +9,17 @@ from tantivy_search_agent import TantivySearchAgent
 load_dotenv()
 
 class SearchAgent:
-    def __init__(self, tantivy_agent: TantivySearchAgent, provider_name: str = "Gemini"):
+    def __init__(self, tantivy_agent: TantivySearchAgent, provider_name: str = "Gemini", api_keys: Dict[str, str] = None):
         """Initialize the search agent with Tantivy agent and LLM client"""
         self.tantivy_agent = tantivy_agent
         self.logger = logging.getLogger(__name__)
         
-        # Initialize LLM provider
-        self.llm_provider = LLMProvider()
+        # Initialize LLM provider with API keys
+        self.llm_provider = LLMProvider(api_keys)
         self.llm = None
         self.set_provider(provider_name)
         
-        self.min_confidence_threshold = 0.5
+        self.min_confidence_threshold = 0.7
 
     def set_provider(self, provider_name: str) -> None:
         self.llm = self.llm_provider.get_provider(provider_name)

app.py

@@ -16,6 +16,7 @@ class SearchAgentUI:
         self.index_path ="./index" # os.getenv("INDEX_PATH", "./index")
         # Google Drive folder ID for the index
         self.gdrive_index_id = os.getenv("GDRIVE_INDEX_ID", "1lpbBCPimwcNfC0VZOlQueA4SHNGIp5_t")
+
     def download_index_from_gdrive(self) -> bool:
         """Download index folder from Google Drive"""
         try:
@@ -30,20 +31,18 @@ class SearchAgentUI:
             with zipfile.ZipFile(zip_path, 'r') as zip_ref:
                 zip_ref.extractall(".")
             
-            # Remove the zip file
-            os.remove(zip_path)
-            return True
+         
         except Exception as e:
             st.error(f"Failed to download index: {str(e)}")
             return False
 
-    def get_available_providers(self) -> List[str]:
+    def get_available_providers(self, api_keys: dict) -> List[str]:
         """Get available providers without creating a SearchAgent instance"""
         temp_tantivy = TantivySearchAgent(self.index_path)
-        temp_agent = SearchAgent(temp_tantivy)
+        temp_agent = SearchAgent(temp_tantivy, api_keys=api_keys)
         return temp_agent.get_available_providers()
 
-    def initialize_system(self):
+    def initialize_system(self, api_keys: dict):
         try:
             # Check if index folder exists
             if not os.path.exists(self.index_path):
@@ -54,10 +53,14 @@ class SearchAgentUI:
 
             self.tantivy_agent = TantivySearchAgent(self.index_path)
             if self.tantivy_agent.validate_index():
-                available_providers = self.get_available_providers()
+                available_providers = self.get_available_providers(api_keys)
+                if not available_providers:
+                    return False, "שגיאה: לא נמצאו ספקי AI זמינים. אנא הזן מפתח API אחד לפחות.", []
+                
                 self.agent = SearchAgent(
                     self.tantivy_agent,
-                    provider_name=st.session_state.get('provider', available_providers[0])
+                    provider_name=st.session_state.get('provider', available_providers[0]),
+                    api_keys=api_keys
                 )
                 return True, "המערכת מוכנה לחיפוש", available_providers
             else:
@@ -69,7 +72,7 @@ class SearchAgentUI:
         st.set_page_config(
             page_title="איתוריא",
             layout="wide",
-            initial_sidebar_state="collapsed"
+            initial_sidebar_state="expanded"
         )
 
         # Enhanced RTL support and styling
@@ -108,50 +111,109 @@ class SearchAgentUI:
                 margin: 5px 0;
                 background-color: white;
             }
+            [data-testid="stSidebar"] {
+                direction: rtl;
+            }
         </style>
         """, unsafe_allow_html=True)
+        
+        st.session_state.api_keys = {
+            'google': "",
+            'openai': "",
+            'anthropic': ""
+            
+        }
 
-        # Initialize system
-        success, status_msg, available_providers = self.initialize_system()
-
-        # Header layout
-        col1, col2, col3 = st.columns([2,1,1])
 
-        with col1:
-            if success:
-                st.success(status_msg)
-            else:
-                st.error(status_msg)
 
-        with col2:
-            if 'provider' not in st.session_state:
-                st.session_state.provider = available_providers[0] if available_providers else None
+        # Sidebar settings
+        with st.sidebar:
+            st.title("הגדרות")
+            
+            # API Key Configuration
+            st.subheader("הגדרת מפתחות API")
+            
+            # Google API Key
+            google_key = st.text_input(
+                "Google API Key",
+                value=st.session_state.api_keys['google'],
+                type="password",                
+                key="google_key",
+                help="הזן את מפתח ה-API של Google Gemini "
+            )
+            st.session_state.api_keys['google'] = google_key
+            
+            st.html('<small> ניתן להשיג מפתח <a href="https://aistudio.google.com/app/apikey">כאן</a> </small>', )
+            
+            # OpenAI API Key
+            openai_key = st.text_input(
+                "OpenAI API Key",
+                value=st.session_state.api_keys['openai'],
+                type="password",
+                key="openai_key",
+                help="הזן את מפתח ה-API של OpenAI"
+            )
+            st.session_state.api_keys['openai'] = openai_key 
             
+                 
+            st.html('<small> ניתן להשיג מפתח <a href="https://platform.openai.com/account/api-keys">כאן</a> </small> ', )
+    
+            
+            # Anthropic API Key
+            anthropic_key = st.text_input(
+                "Anthropic API Key",
+                value=st.session_state.api_keys['anthropic'],
+                type="password",
+                key="anthropic_key",
+                help="הזן את מפתח ה-API של Anthropic Claude"
+            )
+            st.session_state.api_keys['anthropic'] = anthropic_key 
+            
+                 
+            st.html('<small> ניתן להשיג מפתח <a href="https://console.anthropic.com/">כאן</a> </small>', )
+    
+
+            st.markdown("---")
+
+        # Initialize system with current API keys
+        success, status_msg, available_providers = self.initialize_system(st.session_state.api_keys)
+
+        # Continue with sidebar settings
+        with st.sidebar:
             if available_providers:
+                if 'provider' not in st.session_state or st.session_state.provider not in available_providers:
+                    st.session_state.provider = available_providers[0] if available_providers else None
+                
                 provider = st.selectbox(
                     "ספק בינה מלאכותית",
                     options=available_providers,
-                    key='provider'
+                    key='provider',
+                    help="בחר את מודל הAI לשימוש (רק מודלים עם מפתח API זמין יוצגו)"
                 )
                 if self.agent:
                     self.agent.set_provider(provider)
 
-        with col3:
-            col3_1, col3_2 = st.columns(2)
-            with col3_1:
-                max_iterations = st.number_input(
-                    "מספר נסיונות מקסימלי",
-                    min_value=1,
-                    value=3,
-                    key='max_iterations'
-                )
-            with col3_2:
-                results_per_search = st.number_input(
-                    "תוצאות לכל חיפוש",
-                    min_value=1,
-                    value=5,
-                    key='results_per_search'
-                )
+            max_iterations = st.number_input(
+                "מספר נסיונות מקסימלי",
+                min_value=1,
+                value=6,
+                key='max_iterations'
+            )
+            
+            results_per_search = st.number_input(
+                "תוצאות לכל חיפוש",
+                min_value=1,
+                value=10,
+                key='results_per_search'
+            )
+
+        # Main content area
+        st.title("איתוריא")
+        
+        if success:
+            st.success(status_msg)
+        else:
+            st.error(status_msg)
 
         # Search input
         query = st.text_input(

llm_providers.py

@@ -90,35 +90,32 @@ class GeminiProvider:
         raise ValueError("Unsupported prompt format")
 
 class LLMProvider:
-    def __init__(self):
+    def __init__(self, api_keys: Dict[str, str] = None):
         self.providers: Dict[str, Any] = {}
-        self._setup_providers()
+        self._setup_providers(api_keys or {})
 
-    def _setup_providers(self):
-        
-           # Google Gemini
-        if google_key := os.getenv('GOOGLE_API_KEY'):
+    def _setup_providers(self, api_keys: Dict[str, str]):
+        # Google Gemini
+        google_key = api_keys.get('google') or os.getenv('GOOGLE_API_KEY')
+        if google_key:
             self.providers['Gemini'] = GeminiProvider(api_key=google_key)
 
-        
         # Anthropic
-        if anthropic_key := os.getenv('ANTHROPIC_API_KEY'):
+        anthropic_key = api_keys.get('anthropic') or os.getenv('ANTHROPIC_API_KEY')
+        if anthropic_key:
             self.providers['Claude'] = ChatAnthropic(
                 api_key=anthropic_key,
                 model_name="claude-3-5-sonnet-20241022",
-                
             )
 
         # OpenAI
-        if openai_key := os.getenv('OPENAI_API_KEY'):
+        openai_key = api_keys.get('openai') or os.getenv('OPENAI_API_KEY')
+        if openai_key:
             self.providers['ChatGPT'] = ChatOpenAI(
                 api_key=openai_key,
                 model_name="gpt-4o-2024-11-20"
             )
 
-     
-       
-
     def get_available_providers(self) -> list[str]:
         """Return list of available provider names"""
         return list(self.providers.keys())