Update main.py

main.py

@@ -97,12 +97,10 @@ def verify_admin(auth_header):
     return uid
 
 # ---------- Dummy Admin Creation on Startup ----------
-
+"""
 def create_dummy_admin():
     
-    admin_email = "rairorr@gmail.com"
-    admin_password = "rai1034R!"
-    
+
     try:
         # Try to get the user if it exists
         admin_user = auth.get_user_by_email(admin_email)
@@ -121,6 +119,7 @@ def create_dummy_admin():
             'created_at': datetime.utcnow().isoformat()
         })
     print(f"Dummy admin ready: {admin_email}")
+    """
 # ---------- Authentication Endpoints ----------
 
 @app.route('/api/auth/signup', methods=['POST'])
@@ -255,13 +254,12 @@ def generate_story_endpoint():
         user_ref = db.reference(f"users/{uid}")
         user_data = user_ref.get() or {}
         current_credits = user_data.get("credits", 0)
-        
         if current_credits < 5:
             return jsonify({'error': 'Insufficient credits. You need at least 5 credits to generate a story.'}), 403
 
         # --- Read Request Data ---
         data = request.form.to_dict()  # For multipart/form-data
-        input_type = data.get('input_type', 'text')  # "text", "pdf", "wiki", "bible", "youtube", "dataframe"
+        input_type = data.get('input_type', 'text')  
         prompt = data.get('prompt')  # For "text" only
         story_type = data.get('story_type', 'free_form')
         style = data.get('style', 'whimsical')
@@ -272,12 +270,19 @@ def generate_story_endpoint():
         if input_type not in ["text", "pdf", "wiki", "bible", "youtube", "dataframe"]:
             return jsonify({'error': 'Unsupported input_type'}), 400
 
+        # Optionally retrieve these fields if relevant
+        wiki_url = data.get("wiki_url")
+        bible_reference = data.get("bible_reference")
+        youtube_url = data.get("youtube_url")
+        ext = data.get("ext")  # For dataframe usage
+
+        # Prepare for story generation
         from stories import (
             generate_story_from_text,
             get_pdf_text,
             get_df,
             generate_story_from_dataframe,
-            generateResponse  # <-- for chart images
+            generateResponse
         )
         story_gen_start = time.time()
         full_story = None
@@ -297,32 +302,26 @@ def generate_story_endpoint():
 
         elif input_type == "dataframe":
             uploaded_file = request.files.get("file")
-            ext = data.get("ext")  # "csv", "xlsx", "xls"
             if not uploaded_file or not ext:
                 return jsonify({'error': 'File and ext are required for dataframe input'}), 400
-
             df = get_df(uploaded_file, ext)
             if df is None:
                 return jsonify({'error': f'Failed to read {ext} file'}), 400
-
             full_story = generate_story_from_dataframe(df, story_type)
 
         elif input_type == "wiki":
-            wiki_url = data.get("wiki_url")
             if not wiki_url:
                 return jsonify({'error': 'wiki_url is required for input_type "wiki"'}), 400
             from stories import generate_story_from_wiki
             full_story = generate_story_from_wiki(wiki_url, story_type)
 
         elif input_type == "bible":
-            bible_reference = data.get("bible_reference")
             if not bible_reference:
                 return jsonify({'error': 'bible_reference is required for input_type "bible"'}), 400
             from stories import generate_story_from_bible
             full_story = generate_story_from_bible(bible_reference, story_type)
 
         elif input_type == "youtube":
-            youtube_url = data.get("youtube_url")
             if not youtube_url:
                 return jsonify({'error': 'youtube_url is required for input_type "youtube"'}), 400
             from stories import generate_story_from_youtube
@@ -349,28 +348,26 @@ def generate_story_endpoint():
         from image_gen import generate_image_with_retry
         from audio_gen import generate_audio
 
-        # If input_type is "dataframe", we have a df for chart generation
+        # If input_type is "dataframe", re-use df for chart generation
         df = None
         if input_type == "dataframe":
             uploaded_file = request.files.get("file")
-            ext = data.get("ext")
-            df = get_df(uploaded_file, ext)  # re-use the same df
+            if uploaded_file and ext:
+                df = get_df(uploaded_file, ext)
 
         # 3) Process each section
         for section_text in sections_raw:
-            # Extract an image prompt between angle brackets
             img_prompt_match = re.search(r"<(.*?)>", section_text)
             img_prompt = img_prompt_match.group(1).strip() if img_prompt_match else section_text[:100]
 
             image_start = time.time()
             image_obj = None
 
-            # If we are dealing with "dataframe", attempt chart generation first
+            # Attempt chart generation if dataframe
             if input_type == "dataframe" and df is not None:
                 try:
-                    chart_str = generateResponse(img_prompt, df)  # returns a Python string or None
+                    chart_str = generateResponse(img_prompt, df)
                     if chart_str and chart_str.startswith("data:image/png;base64,"):
-                        # decode base64 -> PIL Image
                         base64_data = chart_str.split(",", 1)[1]
                         chart_bytes = base64.b64decode(base64_data)
                         image_obj = Image.open(io.BytesIO(chart_bytes))
@@ -411,6 +408,23 @@ def generate_story_endpoint():
         # 4) Store the story
         story_id = str(uuid.uuid4())
         story_ref = db.reference(f"stories/{story_id}")
+
+        # ---------- STORE INPUT PARAMS -----------
+        input_params = {
+            "input_type": input_type,
+            "prompt": prompt,          # might be None if pdf/dataframe
+            "wiki_url": wiki_url,      # might be None
+            "bible_reference": bible_reference,
+            "youtube_url": youtube_url,
+            "story_type": story_type,
+            "style": style,
+            "voice_model": voice_model,
+            "image_model": image_model,
+            "audio_model": audio_model,
+            "ext": ext                 # for dataframe
+        }
+        # -----------------------------------------
+
         story_record = {
             "uid": uid,
             "full_story": full_story,
@@ -422,7 +436,8 @@ def generate_story_endpoint():
             },
             "created_at": datetime.utcnow().isoformat(),
             "input_type": input_type,
-            "story_type": story_type
+            "story_type": story_type,
+            "input_params": input_params  # <-- store them here
         }
         story_ref.set(story_record)
 
@@ -438,7 +453,8 @@ def generate_story_endpoint():
             "preview": preview,
             "sections": sections,
             "generation_times": story_record["generation_times"],
-            "new_credits": new_credits
+            "new_credits": new_credits,
+            "input_params": input_params  # Return them if you want
         })
 
     except Exception as e:
@@ -1162,5 +1178,5 @@ def admin_update_credits(uid):
 # ---------- Main ----------
 if __name__ == '__main__':
     # Create dummy admin account if it doesn't exist
-    create_dummy_admin()
+    #create_dummy_admin()
     app.run(debug=True, host="0.0.0.0", port=7860)