Create app.py

app.py

@@ -0,0 +1,63 @@
+import os
+from groq import Groq
+import streamlit as st
+from dotenv import load_dotenv
+
+# Load API key from .env file
+load_dotenv()
+api_key = os.getenv("GROQ_API_KEY")
+
+# Initialize the Groq client
+client = Groq(api_key=api_key)
+
+# Define the health issues for the chatbot
+health_issues = [
+    "fever", "malaria", "skin infection", "flu", "cough", "diabetes", "hypertension", 
+    "cold", "asthma", "arthritis", "chickenpox", "measles", "pneumonia", "migraine", 
+    "anxiety", "depression", "heart disease", "cancer", "high cholesterol", "stroke", 
+    "obesity", "insomnia", "constipation", "acid reflux", "eczema", "psoriasis", 
+    "allergy", "cold sores", "chronic fatigue", "UTI", "kidney disease", "hepatitis", 
+    "tuberculosis", "stomach ulcers", "gout", "HIV/AIDS", "malnutrition", "blood pressure", 
+    "menstrual irregularities", "pregnancy", "anemia", "vitamin deficiency", "chronic pain"
+]
+
+
+# Function to fetch chatbot completion from Groq API
+def get_response(query):
+    completion = client.chat.completions.create(
+        model="llama-3.3-70b-versatile",
+        messages=[{"role": "user", "content": query}],
+        temperature=0.7,
+        max_completion_tokens=1024,
+        top_p=1,
+    )
+    response = completion.choices[0].message.content
+    return response
+
+def main():
+    st.title("Health Assistant Chatbot")
+
+    # Let the user choose a health topic or type a custom query
+    topic = st.selectbox("Choose a health issue", health_issues)
+    user_input = st.text_area("Or ask a health-related question:", "")
+
+    # If the user provides a query, we use that
+    query = user_input if user_input else f"Tell me about {topic}"
+
+    # Create a submit button
+    submit_button = st.button("Submit")
+
+    # Call the Groq API to get the response if the button is clicked
+    if submit_button and query:
+        response = get_response(query)
+        
+        # Display the response
+        st.write("### Response:")
+        st.write(response)
+
+    # Handle unrelated queries
+    if user_input and not any(health_issue in user_input.lower() for health_issue in health_issues):
+        st.write("Sorry, I can only answer health-related questions.")
+
+if __name__ == "__main__":
+    main()