Update app.py

app.py

@@ -6,18 +6,36 @@ from PIL import Image
 import io
 from langchain_community.document_loaders import YoutubeLoader
 import pyttsx3
-
+import requests
 # Initialize OpenAI API
-openai.api_key = 'sk-proj-Mavt8kPfsNNsFZAEadbuT3BlbkFJMuSDnGP7gqXIbcLgFS1z'
+openai.api_key = 'sk-proj-vwPU5Rhnd9FE5KO25fNdT3BlbkFJVrEvfXZ9YxxsO5lYzLw3'
 
 # Function to generate content using ChatGPT
 def generate_content(messages):
-    response = openai.ChatCompletion.create(
-        model="gpt-4",
-        messages=messages,
-        max_tokens=1500  # Increase max tokens if needed for more detailed responses
-    )
-    return response.choices[0].message['content'].strip()
+    endpoint = 'https://api.together.xyz/v1/chat/completions'
+    headers = {
+        "Authorization": "Bearer dc073523ec85f07124dd84b16504a472905c1fac5542f3bf43f9043b8162afc7",
+        "Content-Type": "application/json"
+    }
+    data = {
+        "model": "meta-llama/Llama-3-8b-chat-hf",
+        "max_tokens": 1024,
+        "temperature": 0.7,
+        "top_p": 0.7,
+        "top_k": 50,
+        "repetition_penalty": 1,
+        "stop": ["<|eot_id|>"],
+        "messages": messages
+    }
+
+    response = requests.post(endpoint, headers=headers, json=data)
+
+    if response.status_code == 200:
+        generated_text = response.json()['choices'][0]['message']['content']
+        return generated_text
+    else:
+        print(f"Error: {response.status_code} - {response.text}")
+        return "Error generating content"
 
 def summarize_youtube_video(video_url, openai_api_key, max_tokens=1500):
     openai.api_key = openai_api_key
@@ -67,18 +85,13 @@ def summarize_youtube_video(video_url, openai_api_key, max_tokens=1500):
     {transcript_text}
     """
 
-    # Use GPT-4 to summarize the transcript
+    # Use gpt-3.5-turbo to summarize the transcript
     messages = [
         {"role": "system", "content": "You are a helpful teaching assistant."},
         {"role": "user", "content": prompt}
     ]
-    response = openai.ChatCompletion.create(
-        model="gpt-4",
-        messages=messages,
-        max_tokens=max_tokens,
-    )
-    summary = response.choices[0].message['content'].strip()
 
+    summary = generate_content(messages)
     return summary
 
 # Function to create a PDF
@@ -130,7 +143,7 @@ st.title("Teaching Assistant Application")
 st.sidebar.header("Options")
 option = st.sidebar.selectbox(
     "Choose an option",
-    ["Generate Test Questions","Youtube Video Summeriser","Generate Project Based Learning", "Create Course Curriculum", "Create Assignments", "Create Activities","Describe and Discuss Image"]
+    ["Generate Test Questions","Youtube Video Summeriser","Generate Project Based Learning", "Create Course Curriculum", "Create Assignments", "Create Activities"]
 )
 
 st.header(option)
@@ -179,7 +192,15 @@ if option == "Generate Test Questions":
             st.subheader("Generated Content")
             st.write(content)
 
-            
+            if st.button("Export to PDF"):
+                pdf_path = "generated_content.pdf"
+                pdf_data = create_pdf(content, pdf_path)
+                st.download_button(
+                    label="Download PDF",
+                    data=pdf_data.getvalue(),
+                    file_name="generated_content.pdf",
+                    mime="application/pdf"
+                )
         else:
             st.warning("Please enter both a topic and a standard.")
       
@@ -190,12 +211,20 @@ if option == "Youtube Video Summeriser" :
     if st.button("Summarize Video"):
         if video_url:
            
-                content = summarize_youtube_video(video_url, 'sk-proj-Mavt8kPfsNNsFZAEadbuT3BlbkFJMuSDnGP7gqXIbcLgFS1z')
+                content = summarize_youtube_video(video_url, 'sk-proj-vwPU5Rhnd9FE5KO25fNdT3BlbkFJVrEvfXZ9YxxsO5lYzLw3')
                 st.subheader("Generated Report")
                 st.session_state.summary = content
                 st.markdown(content)
 
-                 
+                if st.button("Export to PDF"):
+                            pdf_path = "generated_content.pdf"
+                            pdf_data = create_pdf(content, pdf_path)
+                            st.download_button(
+                                label="Download PDF",
+                                data=pdf_data.getvalue(),
+                                file_name="generated_content.pdf",
+                                mime="application/pdf"
+                            )     
                                  
 if option == "Generate Project Based Learning":
     topic = st.text_input("Enter the topic")
@@ -243,7 +272,16 @@ Be creative, innovative, and engage deeply with the subject matter to make the m
             st.subheader("Generated Content")
             st.write(content)
 
-            
+            if st.button("Export to PDF"):
+                pdf_path = os.path.join("generated_content.pdf")
+                create_pdf(content, pdf_path)
+                with open(pdf_path, "rb") as pdf_file:
+                    st.download_button(
+                        label="Download PDF",
+                        data=pdf_file,
+                        file_name="generated_content.pdf",
+                        mime="application/pdf"
+                    )
         else:
             st.warning("Please enter both a topic and a standard.")
 
@@ -286,7 +324,16 @@ elif option == "Create Course Curriculum":
             st.subheader("Generated Content")
             st.write(content)
 
-            
+            if st.button("Export to PDF"):
+                pdf_path = os.path.join("generated_content.pdf")
+                create_pdf(content, pdf_path)
+                with open(pdf_path, "rb") as pdf_file:
+                    st.download_button(
+                        label="Download PDF",
+                        data=pdf_file,
+                        file_name="generated_content.pdf",
+                        mime="application/pdf"
+                    )
         else:
             st.warning("Please enter the subject(s).")
 
@@ -323,7 +370,16 @@ elif option == "Create Assignments":
             st.subheader("Generated Content")
             st.write(content)
 
-            
+            if st.button("Export to PDF"):
+                pdf_path = os.path.join("generated_content.pdf")
+                create_pdf(content, pdf_path)
+                with open(pdf_path, "rb") as pdf_file:
+                    st.download_button(
+                        label="Download PDF",
+                        data=pdf_file,
+                        file_name="generated_content.pdf",
+                        mime="application/pdf"
+                    )
         else:
             st.warning("Please enter the topic/keywords/goal.")
 
@@ -360,7 +416,16 @@ elif option == "Create Activities":
             st.subheader("Generated Content")
             st.write(content)
 
-            
+            if st.button("Export to PDF"):
+                pdf_path = os.path.join("generated_content.pdf")
+                create_pdf(content, pdf_path)
+                with open(pdf_path, "rb") as pdf_file:
+                    st.download_button(
+                        label="Download PDF",
+                        data=pdf_file,
+                        file_name="generated_content.pdf",
+                        mime="application/pdf"
+                    )
         else:
             st.warning("Please enter the topic/keywords.")
 
@@ -391,4 +456,13 @@ elif option == "Describe and Discuss Image":
             st.subheader("Generated Content")
             st.write(content)
 
-            
+            if st.button("Export to PDF"):
+                pdf_path = os.path.join("generated_content.pdf")
+                create_pdf(content, pdf_path)
+                with open(pdf_path, "rb") as pdf_file:
+                    st.download_button(
+                        label="Download PDF",
+                        data=pdf_file,
+                        file_name="generated_content.pdf",
+                        mime="application/pdf"
+                    )