Update content.py

content.py

@@ -1,106 +1,106 @@
-import requests
-import gradio as gr
-
-from helper import generate_access_token
-
-def generate_content(grade, subject, topics, learning_outcomes, notes, stories, activities, group_dicussion, extras):
-    
-    data = {
-        "institution_id": "inst789",
-        "teacher_id": "teacher456",
-        "grade": grade,
-        "subject": subject,
-        "topics": topics.split(", "),
-        "learning_outcomes": learning_outcomes.split(", "),
-        "content_type": {
-            "notes": notes,
-            "stories": stories,
-            "activities": activities,
-            "group_dicussion": group_dicussion
-        },
-        "extras": extras.split(", ")
-    }
-
-    access_token = generate_access_token()
-
-    if access_token is None:
-        return {"Error": "Failed to generate access token"}
-
-    response = requests.post("http://20.193.151.200:8080/k12/generate/content", 
-                             headers={
-                                 "accept": "application/json",
-                                 "content-type": "application/json",
-                                 "Authorization": f"{access_token}"},
-                             json=data)
-    
-    if(str(response.status_code)[0] != '2'):
-        return {"Error": f"{response.status_code}"}
-
-    return response.json()
-
-def get_content_generation(request_id):
-    
-    access_token = generate_access_token()
-
-    if access_token is None:
-        return {"Error": "Failed to generate access token"}
-
-    url = f"http://20.193.151.200:8080/k12/generate/content/{request_id}"
-    headers = {"accept": "application/json",
-               "Authorization": access_token}
-    
-    response = requests.get(url, headers=headers)
-    
-    if str(response.status_code)[0] == "2":
-        return response.json()
-    else:
-        return {"Error" : f"{response.status_code}"}
-
-def post_interface():
-    
-    with gr.Blocks() as post_page:    
-        grade = gr.Textbox(label="Grade", value="8")
-        subject = gr.Textbox(label="Subject", value="Science")
-
-        topics = gr.Textbox(label="Topics (comma-separated)", value="Plant and Animal Cells, Reproduction in Humans and Animals")
-        learning_outcomes = gr.Textbox(label="Learning Outcomes (comma-separated)")
-        
-        notes = gr.Radio(label="Notes", choices=["Yes", "No"], value="No")
-        stories = gr.Radio(label="Stories", choices=["Yes", "No"], value="No")
-        activities = gr.Radio(label="Activities", choices=["Yes", "No"], value="No")
-        group_dicussion = gr.Radio(label="Group Discussion", choices=["Yes", "No"], value="No")
-        
-        extras = gr.Textbox(label="Extras  (comma-separated)")
-        
-        submit_button = gr.Button("Invoke Request")
-        output = gr.JSON(label="Content Generation ID")
-
-        submit_button.click(
-            fn=generate_content, 
-            inputs=[grade, subject, topics, learning_outcomes, notes, stories, activities, group_dicussion, extras], 
-            outputs=output,
-        )
-
-    return post_page
-
-def get_interface():
-    with gr.Blocks() as get_page:
-
-        interface = gr.Interface(
-            fn=get_content_generation, 
-            inputs=gr.Textbox(label="Enter Request ID"), 
-            outputs="json",
-        )
-        
-    return get_page
-
-def content_generation():
-    gr.Markdown("# Content Generation")
-    with gr.Blocks() as content_generation:
-        with gr.Tabs():
-            with gr.TabItem("POST"):
-                post_interface()
-            with gr.TabItem("GET"):
-                get_interface()
-
+import requests
+import gradio as gr
+
+from helper import generate_access_token
+
+def generate_content(grade, subject, topics, learning_outcomes, notes, stories, activities, group_dicussion, extras):
+    
+    data = {
+        "institution_id": "inst789",
+        "teacher_id": "teacher456",
+        "grade": grade,
+        "subject": subject,
+        "topics": topics.split(", "),
+        "learning_outcomes": learning_outcomes.split(", "),
+        "content_type": {
+            "notes": notes,
+            "stories": stories,
+            "activities": activities,
+            "group_dicussion": group_dicussion
+        },
+        "extras": extras.split(", ")
+    }
+
+    access_token = generate_access_token()
+
+    if access_token is None:
+        return {"Error": "Failed to generate access token"}
+
+    response = requests.post("http://20.193.151.200:8080/v1/k12/generate/content", 
+                             headers={
+                                 "accept": "application/json",
+                                 "content-type": "application/json",
+                                 "Authorization": f"{access_token}"},
+                             json=data)
+    
+    if(str(response.status_code)[0] != '2'):
+        return {"Error": f"{response.status_code}"}
+
+    return response.json()
+
+def get_content_generation(request_id):
+    
+    access_token = generate_access_token()
+
+    if access_token is None:
+        return {"Error": "Failed to generate access token"}
+
+    url = f"http://20.193.151.200:8080/k12/generate/content/{request_id}"
+    headers = {"accept": "application/json",
+               "Authorization": access_token}
+    
+    response = requests.get(url, headers=headers)
+    
+    if str(response.status_code)[0] == "2":
+        return response.json()
+    else:
+        return {"Error" : f"{response.status_code}"}
+
+def post_interface():
+    
+    with gr.Blocks() as post_page:    
+        grade = gr.Textbox(label="Grade", value="8")
+        subject = gr.Textbox(label="Subject", value="Science")
+
+        topics = gr.Textbox(label="Topics (comma-separated)", value="Plant and Animal Cells, Reproduction in Humans and Animals")
+        learning_outcomes = gr.Textbox(label="Learning Outcomes (comma-separated)")
+        
+        notes = gr.Radio(label="Notes", choices=["Yes", "No"], value="No")
+        stories = gr.Radio(label="Stories", choices=["Yes", "No"], value="No")
+        activities = gr.Radio(label="Activities", choices=["Yes", "No"], value="No")
+        group_dicussion = gr.Radio(label="Group Discussion", choices=["Yes", "No"], value="No")
+        
+        extras = gr.Textbox(label="Extras  (comma-separated)")
+        
+        submit_button = gr.Button("Invoke Request")
+        output = gr.JSON(label="Content Generation ID")
+
+        submit_button.click(
+            fn=generate_content, 
+            inputs=[grade, subject, topics, learning_outcomes, notes, stories, activities, group_dicussion, extras], 
+            outputs=output,
+        )
+
+    return post_page
+
+def get_interface():
+    with gr.Blocks() as get_page:
+
+        interface = gr.Interface(
+            fn=get_content_generation, 
+            inputs=gr.Textbox(label="Enter Request ID"), 
+            outputs="json",
+        )
+        
+    return get_page
+
+def content_generation():
+    gr.Markdown("# Content Generation")
+    with gr.Blocks() as content_generation:
+        with gr.Tabs():
+            with gr.TabItem("POST"):
+                post_interface()
+            with gr.TabItem("GET"):
+                get_interface()
+
     return content_generation