commented app.py

app.py

@@ -1,29 +1,46 @@
-import os
-import requests
-from transformers import pipeline
-import gradio as gr
+# import os
+# import requests
+# from transformers import pipeline
+# import gradio as gr
 
-# Use the token from environment variable (secret)
-token = os.environ.get("OPENAI_API_KEY")
-# generator = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta", token=token)
+# # Use the token from environment variable (secret)
+# token = os.environ.get("OPENAI_API_KEY")
+# # generator = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta", token=token)
 
-# API_URL = "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta"
-API_URL = "https://api-inference.huggingface.co/models/tiiuae/falcon-rw-1b"
-headers = {"Authorization": f"Bearer {token}"}
+# # API_URL = "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta"
+# API_URL = "https://api-inference.huggingface.co/models/tiiuae/falcon-rw-1b"
+# headers = {"Authorization": f"Bearer {token}"}
 
-# generator = pipeline("text-generation", model="tiiuae/falcon-rw-1b")
+# # generator = pipeline("text-generation", model="tiiuae/falcon-rw-1b")
 
-def query(payload):
-    response = requests.post(API_URL, headers=headers, json=payload)
-    return response.json()
+# def query(payload):
+#     response = requests.post(API_URL, headers=headers, json=payload)
+#     return response.json()
 
-def clean_topic(topic):
-    topic = topic.lower()
-    if "write a blog on" in topic:
-        topic = topic.replace("write a blog on", "").strip()
-    elif "write a blog about" in topic:
-        topic = topic.replace("write a blog about", "").strip()
-    return topic.capitalize()
+# def clean_topic(topic):
+#     topic = topic.lower()
+#     if "write a blog on" in topic:
+#         topic = topic.replace("write a blog on", "").strip()
+#     elif "write a blog about" in topic:
+#         topic = topic.replace("write a blog about", "").strip()
+#     return topic.capitalize()
+
+# # def generate_blog(topic):
+# #     topic = clean_topic(topic)
+# #     prompt = f"""
+# #     Write a detailed and engaging blog post about "{topic}".
+# #     Include an introduction, 2–3 subheadings with paragraphs, and a conclusion.
+# #     Make it informative and conversational.
+# #     """
+# #     # output = query({"inputs": prompt})
+# #     # return output[0]["generated_text"]
+# #     # result = generator(prompt, max_length=700, do_sample=True, temperature=0.7, top_p=0.9)
+# #     # result = generator(prompt, max_length=300, do_sample=True, temperature=0.7, top_p=0.9)
+
+# #     # return result[0]['generated_text']
+
+# #     output = query({"inputs": prompt})
+# #     return output[0]["generated_text"]
 
 # def generate_blog(topic):
 #     topic = clean_topic(topic)
@@ -32,38 +49,21 @@ def clean_topic(topic):
 #     Include an introduction, 2–3 subheadings with paragraphs, and a conclusion.
 #     Make it informative and conversational.
 #     """
-#     # output = query({"inputs": prompt})
-#     # return output[0]["generated_text"]
-#     # result = generator(prompt, max_length=700, do_sample=True, temperature=0.7, top_p=0.9)
-#     # result = generator(prompt, max_length=300, do_sample=True, temperature=0.7, top_p=0.9)
-
-#     # return result[0]['generated_text']
-
 #     output = query({"inputs": prompt})
-#     return output[0]["generated_text"]
-
-def generate_blog(topic):
-    topic = clean_topic(topic)
-    prompt = f"""
-    Write a detailed and engaging blog post about "{topic}".
-    Include an introduction, 2–3 subheadings with paragraphs, and a conclusion.
-    Make it informative and conversational.
-    """
-    output = query({"inputs": prompt})
-    print("API Response:", output)  # <-- Add this for debugging
+#     print("API Response:", output)  # <-- Add this for debugging
 
-    if isinstance(output, list) and "generated_text" in output[0]:
-        return output[0]["generated_text"]
-    elif "error" in output:
-        return f"Error from model: {output['error']}"
-    else:
-        return "Failed to generate blog. Please try again."
+#     if isinstance(output, list) and "generated_text" in output[0]:
+#         return output[0]["generated_text"]
+#     elif "error" in output:
+#         return f"Error from model: {output['error']}"
+#     else:
+#         return "Failed to generate blog. Please try again."
 
 
-gr.Interface(
-    fn=generate_blog,
-    inputs="text",
-    outputs="text",
-    title="AI Blog Writer"
-).launch(share=True)
+# gr.Interface(
+#     fn=generate_blog,
+#     inputs="text",
+#     outputs="text",
+#     title="AI Blog Writer"
+# ).launch(share=True)