Update app.py

app.py

@@ -1,82 +1,122 @@
 import streamlit as st
 import requests
+import google.generativeai as palm
 
 # Set the page title and icon
-st.set_page_config(page_title="Wikipedia Microbot", page_icon=":mag:")
-
-# Define a CSS style for better layout
-st.markdown(
-    """
-    <style>
-    .stTextInput, .stButton, .stProgress { margin: 10px 0; }
-    .stButton { width: 100%; }
-    .stAlert { padding: 10px; }
-    .stMarkdown img { max-width: 100%; }
-    .stText { font-weight: bold; }
-    </style>
-    """,
-    unsafe_allow_html=True
-)
+st.set_page_config(page_title="Wikipedia Microbot and AI Mode")
 
+# Wikipedia Microbot Constants
 WIKIPEDIA_API_URL = "https://en.wikipedia.org/w/api.php"
 
-st.title("Wikipedia Microbot")
-
-# Create a sidebar for better organization
-st.sidebar.header("Options")
-
-# Create widgets in the sidebar
-query = st.sidebar.text_input("Enter a Query")
-search_button = st.sidebar.button("Search")
-clear_output_button = st.sidebar.button("Clear Output")
-
-# Create a container for the main content
-main_container = st.container()
-
-if search_button:
-    if query:
-        try:
-            # Search Wikipedia for the query
-            params = {
-                "action": "query",
-                "format": "json",
-                "prop": "extracts|images|info|pageviews",
-                "exintro": True,
-                "explaintext": True,
-                "exsectionformat": "plain",
-                "titles": query,
-                "utf8": 1,
-                "formatversion": 2,
-                "pvipdays": 7,  # Get page views statistics for the last 7 days
-            }
-
-            response = requests.get(WIKIPEDIA_API_URL, params=params)
-
-            if response.status_code == 200:
-                data = response.json()
-
-                if "error" in data:
-                    st.sidebar.error(f"Error: {data['error']['info']}")
+# AI Mode Constants
+palm.configure(api_key="AIzaSyCnrcQWTEedcGuEsJ9Uhbj_ZS9B2nozMiU")
+defaults = {
+    'model': 'models/text-bison-001',
+    'temperature': 1,
+    'candidate_count': 1,
+    'top_k': 40,
+    'top_p': 0.95,
+    'max_output_tokens': 1024,
+    'stop_sequences': [],
+    'safety_settings': [
+        {"category": "HARM_CATEGORY_DEROGATORY", "threshold": 4},
+        {"category": "HARM_CATEGORY_TOXICITY", "threshold": 4},
+        {"category": "HARM_CATEGORY_VIOLENCE", "threshold": 4},
+        {"category": "HARM_CATEGORY_SEXUAL", "threshold": 4},
+        {"category": "HARM_CATEGORY_MEDICAL", "threshold": 4},
+        {"category": "HARM_CATEGORY_DANGEROUS", "threshold": 4},
+    ]
+}
+
+# Custom Instruction
+custom_instruction = """
+I am a small system from HelpingAI made by Abhay Koul who is 16 years old and lives in KP Colony, Vessu, Anantnag. Abhay Koul is currently working on more AI and AI-related projects.Certainly! If you're looking for more information about Abhay Koul's projects and HelpingAI, I would recommend joining the Discord community. You can find additional details there. He has been working on AI from Jan 2023. If you want more details, you can join the Discord community. The main function of HelpingAI is to make learning and education easier and more convenient for students. My purpose is to simplify your learning journey by providing personalized assistance, innovative teaching methods, and tailored resources to meet your unique needs. I am here to make your educational experience more enjoyable and effective. Feel free to ask me any questions or let me know how I can assist you in your learning adventure and in many more things from your life.
+"""
+
+# Create tabs for Wikipedia Microbot and AI Mode
+selected_mode = st.radio("Select Mode", ["Wikipedia Microbot", "AI Mode"])
+
+if selected_mode == "Wikipedia Microbot":
+    # Wikipedia Microbot Code
+    st.title("Wikipedia Microbot")
+    st.markdown("Explore Wikipedia with Ease")
+
+    # Sidebar for user options
+    st.sidebar.header("Options")
+
+    # User input and search button
+    query = st.sidebar.text_input("Enter a Query", help="E.g., 'Python programming'")
+    search_button = st.sidebar.button("Search")
+
+    # Container for main content
+    main_container = st.container()
+
+    if search_button:
+        if query:
+            try:
+                # Search Wikipedia for the query
+                params = {
+                    "action": "query",
+                    "format": "json",
+                    "prop": "extracts|info|pageviews",
+                    "exintro": True,
+                    "explaintext": True,
+                    "exsectionformat": "plain",
+                    "titles": query,
+                    "utf8": 1,
+                    "formatversion": 2,
+                    "pvipdays": 7,
+                }
+
+                response = requests.get(WIKIPEDIA_API_URL, params=params)
+
+                if response.status_code == 200:
+                    data = response.json()
+
+                    if "error" in data:
+                        st.sidebar.error(f"Error: {data['error']['info']}")
+                    else:
+                        page = data["query"]["pages"][0]
+
+                        # Display page title
+                        st.title(page['title'])
+
+                        # Display page views statistics
+                        views = page.get("pageviews", {}).get(query, "Data not available")
+                        st.info(f"Page Views (Last 7 days): {views}")
+
+                        # Display summary
+                        st.write(page.get("extract", "No summary available."))
+
                 else:
-                    page = data["query"]["pages"][0]
+                    st.sidebar.error("Error: Unable to retrieve data from Wikipedia. Please try again later.")
+            except Exception as e:
+                st.sidebar.error(f"Error: {e}")
 
-                    summary = page.get("extract", "")
-                    html_content = ""
-                    for image in page.get("images", []):
-                        image_url = f"https://en.wikipedia.org/wiki/{image['title']}"
-                        # Display the images using st.image
-                        st.sidebar.image(image_url, caption=image['title'], use_column_width=True)
+elif selected_mode == "AI Mode":
+    # AI Mode Code
+    st.title("AI Mode")
+    st.markdown("Interact with an AI powered by Abhay Koul")
 
-                    # Display page views statistics
-                    views = page.get("pageviews", {}).get(query, "Data not available")
-                    views_text = f"Page Views (Last 7 days): {views}"
+    user_input = st.text_area('You:', height=100, help="Type your message here")
 
-                    result = f"# {page['title']}\n{views_text}\n{summary}\n{html_content}"
-                    main_container.markdown(result, unsafe_allow_html=True)
+    if st.button('Submit', key='ai_button'):
+        with st.spinner("Thinking..."):
+            if user_input.lower() in ['quit', 'exit', 'bye']:
+                st.write("Goodbye! Have a great day!")
             else:
-                st.sidebar.error("Error: Unable to retrieve data from Wikipedia. Please try again later.")
-        except Exception as e:
-            st.sidebar.error(f"Error: {e}")
+                # Create a chat history session state
+                session_state = st.session_state.get(user_input, [])
+                session_state.append({"user": user_input})
+                st.session_state[user_input] = session_state
+
+                # Prepare conversation history
+                conversation_history = "\n".join(["You: " + item["user"] for item in session_state])
+
+                # Construct the prompt with conversation history
+                prompt = f"""{custom_instruction}
+Your conversation history:\n{conversation_history}
+Your AI-generated response:"""
 
-if clear_output_button:
-    main_container.text("Output Cleared")
+                response = palm.generate_text(**defaults, prompt=prompt)
+                st.write(response.result)