2ND

.env

@@ -0,0 +1 @@
+AIzaSyBbzFCa84gRACICF9JrjGtonTl8UIdNOPs

app.py

@@ -4,25 +4,22 @@ from PyPDF2 import PdfReader
 from PIL import Image
 import re
 from collections import Counter
-import google.generativeai as genai
+from huggingface_hub import InferenceClient
 from streamlit_option_menu import option_menu
 
-# Get the API key from environment variable
+# Get the Hugging Face API token from environment variable
 api_key = os.getenv("GEMINI_API_KEY")
 
 if api_key is None:
-    st.error("API key not found. Please set the GEMINI_API_KEY environment variable.")
+    st.error("API key not found. Please set the HF_API_KEY environment variable.")
 else:
-    # Gemini Model Initialization
-    MODEL_ID = "gemini-2.0-flash-exp"
-    genai.configure(api_key=api_key)
-    model = genai.GenerativeModel(MODEL_ID)
-    chat = model.start_chat()
+    # Initialize Hugging Face Inference Client
+    client = InferenceClient(api_key=api_key)
 
     st.set_page_config(
         page_title="AI-Powered Historical Document Analysis",
         layout="wide",
-        page_icon="📜"
+        page_icon=":scroll:"  # Updated to avoid UnicodeEncodeError
     )
 
     # Center content and enhance visual style
@@ -42,6 +39,12 @@ else:
             h1, h2, h3, h4, h5 {
                 color: #2c3e50;
             }
+            .nav-label {
+                font-size: 16px;
+                font-weight: bold;
+                margin-bottom: 5px;
+                text-align: center;
+            }
         </style>
     """, unsafe_allow_html=True)
 
@@ -54,17 +57,18 @@ else:
         to extract key information, provide contextual insights, and visualize data on modern maps.
         """)
 
-    # Centered Navigation
+    # Compact Navigation
+    st.markdown("<div class='nav-label'>Navigation</div>", unsafe_allow_html=True)
     selected_tab = option_menu(
-        menu_title="Navigation",
+        menu_title="",
         options=["Home", "Key Points", "General Contents", "Historical Context"],
         icons=["house", "key", "book", "clock"],
         menu_icon="cast",
         default_index=0,
         orientation="horizontal",
         styles={
-            "container": {"display": "flex", "justify-content": "center", "align-items": "center"},
-            "nav-link": {"font-size": "18px", "text-align": "center", "--hover-color": "#f0f0f0"},
+            "container": {"display": "flex", "justify-content": "center", "align-items": "center", "gap": "10px"},
+            "nav-link": {"font-size": "14px", "text-align": "center", "padding": "5px 10px", "--hover-color": "#f0f0f0"},
             "nav-link-selected": {"background-color": "#007bff", "color": "white", "border-radius": "5px"}
         }
     )
@@ -91,9 +95,8 @@ else:
     def contextualize_document(text):
         """Generate historical context based on document text."""
         try:
-            prompt = f"Provide historical context for the following text:\n\n{text[:1000]}"
-            response = chat.send_message(prompt)
-            return response.text
+            response = client.text_generation(model="gpt-neo-2.7B", inputs=f"Provide historical context for the following text:\n\n{text[:1000]}")
+            return response["generated_text"]
         except Exception as e:
             return f"Error: {e}"