Update app.py

app.py

@@ -66,12 +66,9 @@ TRANSLATIONS = {
 
 def translate_text(text):
     if st.session_state.get("language", "English") == "Marathi":
-        if text == "Krushi Mitra":
-            return "कृषी मित्र"
         return TRANSLATIONS.get(text, text)
     return text
 
-    
 # Custom CSS for styling
 # Update custom CSS
 custom_css = """
@@ -194,7 +191,6 @@ def get_plant_info(disease, plant_type="Unknown"):
     - Treatment & Prevention
     """
     try:
-        # API_URL = "http://192.168.56.1:1234/v1/chat/completions"
         API_URL = "https://api-inference.huggingface.co/models/meta-llama/Llama-3.2-1B-Instruct/v1/chat/completions"
         headers = {"Authorization": f"Bearer {os.getenv('HUGGINGFACE_API_TOKEN')}"}
         try:
@@ -204,8 +200,7 @@ def get_plant_info(disease, plant_type="Unknown"):
                 {"role": "user", "content": prompt},
             ]
         }
-            # response = requests.post(API_URL, headers=headers, json=data)
-            response = requests.post(API_URL, json=data)
+            response = requests.post(API_URL, headers=headers, json=data)
             response.raise_for_status()
             print("Response:", data)
         except requests.exceptions.HTTPError as http_err:
@@ -225,8 +220,8 @@ def get_web_pesticide_info(disease, plant_type="Unknown"):
     query = f"site:agrowon.esakal.com {disease} in {plant_type}"
     url = "https://www.googleapis.com/customsearch/v1"
     params = {
-        "key": "AIzaSyCp9j5OGZb5hlykMIAJhbDII3IHYJWCrnQ",
-        "cx": "41b10936a10424ac2",
+        "key": os.getenv("GOOGLE_API_KEY"),
+        "cx": os.getenv("GOOGLE_CX"),
         "q": query,
         "num": 3
     }
@@ -247,8 +242,8 @@ def get_web_pesticide_info(disease, plant_type="Unknown"):
 def get_more_web_info(query):
     url = "https://www.googleapis.com/customsearch/v1"
     params = {
-        "key": "AIzaSyCp9j5OGZb5hlykMIAJhbDII3IHYJWCrnQ",
-        "cx": "41b10936a10424ac2",
+        "key": os.getenv("GOOGLE_API_KEY"),
+        "cx": os.getenv("GOOGLE_CX"),
         "q": query,
         "num": 3
     }
@@ -293,25 +288,16 @@ def translate_text(text):
 
 
 def main():
-    # Initialize language in session state if not present
-    if "language" not in st.session_state:
-        st.session_state.language = "English"
-    
     st.sidebar.title(translate_text("Settings"))
     st.sidebar.info(translate_text("Choose language and plant type, then upload an image to classify the disease."))
-    
-    # Update language state only when radio button changes
-    new_language = st.sidebar.radio(translate_text("Language"), options=["English", "Marathi"], index=["English", "Marathi"].index(st.session_state.language))
-    if new_language != st.session_state.language:
-        st.session_state.language = new_language
-    
+    language_option = st.sidebar.radio(translate_text("Language"), options=["English", "Marathi"], index=0)
+    st.session_state.language = language_option
     plant_type = st.sidebar.selectbox(translate_text("Select Plant Type"), options=['sugarcane', 'maize', 'cotton', 'rice', 'wheat'])
     uploaded_file = st.sidebar.file_uploader(
         translate_text("Upload a plant image..."), 
         type=["jpg", "jpeg", "png"],
         help=translate_text("Select an image of your plant to detect diseases")
     )
-    
     st.title(translate_text("Krushi Mitra"))
     st.write(translate_text("Plant Disease Classification and Pesticide Recommendation.\n\nUpload an image, select plant type, and click on Classify."))
 
@@ -338,8 +324,8 @@ def main():
                     web_recommendation = get_web_pesticide_info(predicted_class, plant_type)
                     if web_recommendation:
                         st.markdown(translate_text("#### Additional Pesticide Recommendations"))
-                        st.markdown(f"**{translate_text('Title')}:** {translate_text(web_recommendation['title'])}")
-                        st.markdown(f"**{translate_text('Summary')}:** {translate_text(web_recommendation['summary'])}")
+                        st.markdown(f"{translate_text('Title')}:** {translate_text(web_recommendation['title'])}")
+                        st.markdown(f"{translate_text('Summary')}:** {translate_text(web_recommendation['summary'])}")
                         if web_recommendation['link']:
                             st.markdown(f"[{translate_text('Read More')}]({web_recommendation['link']})")
                     else:
@@ -349,8 +335,8 @@ def main():
                         commercial_products = get_commercial_product_info(pesticide)
                     if commercial_products:
                         for item in commercial_products:
-                            st.markdown(f"**{translate_text('Title')}:** {translate_text(item['title'])}")
-                            st.markdown(f"**{translate_text('Snippet')}:** {translate_text(item['snippet'])}")
+                            st.markdown(f"{translate_text('Title')}:** {translate_text(item['title'])}")
+                            st.markdown(f"{translate_text('Snippet')}:** {translate_text(item['snippet'])}")
                             if item['link']:
                                 st.markdown(f"[{translate_text('Read More')}]({item['link']})")
                             st.markdown("---")
@@ -361,8 +347,8 @@ def main():
                         more_info = get_more_web_info(f"{predicted_class} in {plant_type}")
                     if more_info:
                         for item in more_info:
-                            st.markdown(f"**{translate_text('Title')}:** {translate_text(item['title'])}")
-                            st.markdown(f"**{translate_text('Snippet')}:** {translate_text(item['snippet'])}")
+                            st.markdown(f"{translate_text('Title')}:** {translate_text(item['title'])}")
+                            st.markdown(f"{translate_text('Snippet')}:** {translate_text(item['snippet'])}")
                             if item['link']:
                                 st.markdown(f"[{translate_text('Read More')}]({item['link']})")
                             st.markdown("---")
@@ -374,4 +360,4 @@ def main():
         st.info(translate_text("Please upload an image from the sidebar to get started."))
 
 if __name__ == "__main__":
-    main()
+    main()