sreepathi-ravikumar commited on
Commit
3eb4e3e
·
verified ·
1 Parent(s): 90613f0

Update text2generation.py

Browse files
Files changed (1) hide show
  1. text2generation.py +18 -13
text2generation.py CHANGED
@@ -1,15 +1,20 @@
1
  import os
2
- from openai import OpenAI
 
 
 
 
 
 
 
 
 
 
3
 
4
  def generate_educational_content(question):
5
  try:
6
- client = OpenAI(
7
- base_url="https://openrouter.ai/api/v1",
8
- api_key=os.getenv("OPENROUTER_API_KEY")
9
- )
10
-
11
  # Step 1: Get detailed explanation
12
- detailed_response = client.chat.completions.create(
13
  model="deepseek/deepseek-chat-v3-0324:free",
14
  messages=[{
15
  "role": "user",
@@ -22,8 +27,8 @@ def generate_educational_content(question):
22
  )
23
  detailed_answer = detailed_response.choices[0].message.content
24
 
25
- # Step 2: Generate video script
26
- video_script = client.chat.completions.create(
27
  model="deepseek/deepseek-chat-v3-0324:free",
28
  messages=[{
29
  "role": "user",
@@ -39,8 +44,8 @@ def generate_educational_content(question):
39
  max_tokens=1500
40
  ).choices[0].message.content
41
 
42
- # Step 3: Generate audio script
43
- audio_script = client.chat.completions.create(
44
  model="deepseek/deepseek-chat-v3-0324:free",
45
  messages=[{
46
  "role": "user",
@@ -52,8 +57,8 @@ def generate_educational_content(question):
52
  max_tokens=1000
53
  ).choices[0].message.content
54
 
55
- # Step 4: Generate summary
56
- summary = client.chat.completions.create(
57
  model="deepseek/deepseek-chat-v3-0324:free",
58
  messages=[{
59
  "role": "user",
 
1
  import os
2
+ import openai
3
+
4
+ # Setup your OpenRouter API key and base
5
+ openai.api_key = os.getenv("OPENROUTER_API_KEY")
6
+ openai.api_base = "https://openrouter.ai/api/v1"
7
+
8
+ # Optional but recommended: Set headers via environment if needed
9
+ # openai._default_headers.update({
10
+ # "HTTP-Referer": "https://sreepathi-ravikumar-backendprocess.hf.space",
11
+ # "X-Title": "Educational AI Assistant"
12
+ # })
13
 
14
  def generate_educational_content(question):
15
  try:
 
 
 
 
 
16
  # Step 1: Get detailed explanation
17
+ detailed_response = openai.ChatCompletion.create(
18
  model="deepseek/deepseek-chat-v3-0324:free",
19
  messages=[{
20
  "role": "user",
 
27
  )
28
  detailed_answer = detailed_response.choices[0].message.content
29
 
30
+ # Step 2: Video script
31
+ video_script = openai.ChatCompletion.create(
32
  model="deepseek/deepseek-chat-v3-0324:free",
33
  messages=[{
34
  "role": "user",
 
44
  max_tokens=1500
45
  ).choices[0].message.content
46
 
47
+ # Step 3: Audio script
48
+ audio_script = openai.ChatCompletion.create(
49
  model="deepseek/deepseek-chat-v3-0324:free",
50
  messages=[{
51
  "role": "user",
 
57
  max_tokens=1000
58
  ).choices[0].message.content
59
 
60
+ # Step 4: Summary
61
+ summary = openai.ChatCompletion.create(
62
  model="deepseek/deepseek-chat-v3-0324:free",
63
  messages=[{
64
  "role": "user",